#include "w25qxx.h"
#include "string.h"

uint16_t W25QXX_TYPE=W25Q16;


//void StartDefaultTask(void const * argument)
//{
//	uint8_t tbuf[]="0123456789";
//	uint8_t rbuf[10];
//	w25q_init();
//	w25q_eraseSector(0x0000);
//	w25q_writeNoCheck(tbuf,0x0000,10);
//	
//  for(;;)
//  {
//		w25q_read(rbuf,0x0000,10);
//		if(memcmp(tbuf,rbuf,10)==0)
//		{
//				HAL_GPIO_TogglePin(D1_GPIO_Port,D1_Pin);
//		}
//    osDelay(100);
//  }
//}

void w25q_delayMs(uint8_t ms)
{
	if(osThreadGetId ()==NULL)HAL_Delay(ms); //不是在os任务中运行就用普通延时
	else osDelay(ms);   //在os任务中运行使用系统延时，避免阻塞其他任务
}

uint8_t w25q_spi_readWriteByte(uint8_t TxData)
{
    uint8_t Rxdata;
    HAL_SPI_TransmitReceive(&hspi2,&TxData,&Rxdata,1, 1000);       
 	return Rxdata;          		    //返回收到的数据		
}

void w25q_init()
{
	uint8_t temp;
	W25QXX_TYPE=w25q_readId();	        //读取FLASH ID.
	if(W25QXX_TYPE==W25Q256)            //SPI FLASH为W25Q256
	{
		temp=w25q_readSr(3);              //读取状态寄存器3，判断地址模式
		if((temp&0X01)==0)			        	//如果不是4字节地址模式,则进入4字节地址模式
		{
			W25Q_CS_LOW(); 			        		//选中
			w25q_spi_readWriteByte(W25X_Enable4ByteAddr);//发送进入4字节地址模式指令   
			W25Q_CS_HIGH();       		      //取消片选   
		}
	}
	else if(W25QXX_TYPE==W25Q16)
	{
		//HAL_GPIO_WritePin(LEDD_GPIO_Port,LEDD_Pin,GPIO_PIN_SET);
	}
}

//读取W25QXX的状态寄存器，W25QXX一共有3个状态寄存器
//状态寄存器1：
//BIT7  6   5   4   3   2   1   0
//SPR   RV  TB BP2 BP1 BP0 WEL BUSY
//SPR:默认0,状态寄存器保护位,配合WP使用
//TB,BP2,BP1,BP0:FLASH区域写保护设置
//WEL:写使能锁定
//BUSY:忙标记位(1,忙;0,空闲)
//默认:0x00
//状态寄存器2：
//BIT7  6   5   4   3   2   1   0
//SUS   CMP LB3 LB2 LB1 (R) QE  SRP1
//状态寄存器3：
//BIT7      6    5    4   3   2   1   0
//HOLD/RST  DRV1 DRV0 (R) (R) WPS ADP ADS
//regno:状态寄存器号，范:1~3
//返回值:状态寄存器值
uint8_t w25q_readSr(uint8_t regno)   
{  
	uint8_t byte=0,command=0; 
	switch(regno)
	{
			case 1:
					command=W25X_ReadStatusReg1;    //读状态寄存器1指令
					break;
			case 2:
					command=W25X_ReadStatusReg2;    //读状态寄存器2指令
					break;
			case 3:
					command=W25X_ReadStatusReg3;    //读状态寄存器3指令
					break;
			default:
					command=W25X_ReadStatusReg1;    
					break;
	}    
	W25Q_CS_LOW();                            	//使能器件   
	w25q_spi_readWriteByte(command);            //发送读取状态寄存器命令    
	byte=w25q_spi_readWriteByte(0Xff);          //读取一个字节  
	W25Q_CS_HIGH();                            	//取消片选     
	return byte;   
} 

//写W25QXX状态寄存器
void w25q_writeSr(uint8_t regno,uint8_t sr)   
{   
	uint8_t command=0;
	switch(regno)
	{
			case 1:
					command=W25X_WriteStatusReg1;    //写状态寄存器1指令
					break;
			case 2:
					command=W25X_WriteStatusReg2;    //写状态寄存器2指令
					break;
			case 3:
					command=W25X_WriteStatusReg3;    //写状态寄存器3指令
					break;
			default:
					command=W25X_WriteStatusReg1;    
					break;
	}   
	W25Q_CS_LOW();                            	//使能器件   
	w25q_spi_readWriteByte(command);            //发送写取状态寄存器命令    
	w25q_spi_readWriteByte(sr);                 //写入一个字节  
	W25Q_CS_HIGH();                            	//取消片选     	      
}

//等待空闲
void w25q_waitBusy(void)   
{   
	while((w25q_readSr(1)&0x01)==0x01)   // 等待BUSY位清空
	{
		w25q_delayMs(1); //带操作系统时防止阻塞其他任务运行
	}
} 

//W25QXX写使能	
//将WEL置位   
void w25q_writeEnable(void)   
{
	W25Q_CS_LOW();                            	//使能器件   
	w25q_spi_readWriteByte(W25X_WriteEnable);   //发送写使能  
	W25Q_CS_HIGH();                            	//取消片选     	      
}
void w25q_writeDisable(void) 
{
	W25Q_CS_LOW();                            	 //使能器件   
	w25q_spi_readWriteByte(W25X_WriteDisable);   //发送写禁止指令  
	W25Q_CS_HIGH();
}

//读取芯片ID
//返回值如下:				   
//0XEF13,表示芯片型号为W25Q80  
//0XEF14,表示芯片型号为W25Q16    
//0XEF15,表示芯片型号为W25Q32  
//0XEF16,表示芯片型号为W25Q64 
//0XEF17,表示芯片型号为W25Q128 	  
//0XEF18,表示芯片型号为W25Q256
uint16_t w25q_readId(void)
{
	uint16_t Temp = 0;	  
	W25Q_CS_LOW();				    
	w25q_spi_readWriteByte(0x90);//发送读取ID命令	    
	w25q_spi_readWriteByte(0x00); 	    
	w25q_spi_readWriteByte(0x00); 	    
	w25q_spi_readWriteByte(0x00); 	 			   
	Temp|=w25q_spi_readWriteByte(0xFF)<<8;  
	Temp|=w25q_spi_readWriteByte(0xFF);	 
	W25Q_CS_HIGH();				    
	return Temp;
} 

//读取SPI FLASH  
//在指定地址开始读取指定长度的数据
//pBuffer:数据存储区
//ReadAddr:开始读取的地址(24bit)
//NumByteToRead:要读取的字节数(最大65535)
void w25q_read(uint8_t* pBuffer,uint32_t ReadAddr,uint16_t NumByteToRead)   
{ 
 	uint16_t i;   										    
	W25Q_CS_LOW();                            //使能器件   
	w25q_spi_readWriteByte(W25X_ReadData);        //发送读取命令  
	if(W25QXX_TYPE==W25Q256)                  //如果是W25Q256的话地址为4字节的，要发送最高8位
	{
			w25q_spi_readWriteByte((uint8_t)((ReadAddr)>>24));    
	}
	w25q_spi_readWriteByte((uint8_t)((ReadAddr)>>16));   //发送24bit地址    
	w25q_spi_readWriteByte((uint8_t)((ReadAddr)>>8));   
	w25q_spi_readWriteByte((uint8_t)ReadAddr);   
	for(i=0;i<NumByteToRead;i++)
	{ 
     pBuffer[i]=w25q_spi_readWriteByte(0XFF);    //循环读数  
  }
	W25Q_CS_HIGH();  				    	      
}  
//SPI在一页(0~65535)内写入少于256个字节的数据
//在指定地址开始写入最大256字节的数据
//pBuffer:数据存储区
//WriteAddr:开始写入的地址(24bit)
//NumByteToWrite:要写入的字节数(最大256),该数不应该超过该页的剩余字节数!!!	 
void w25q_writePage(uint8_t* pBuffer,uint32_t WriteAddr,uint16_t NumByteToWrite)
{
 	uint16_t i;  
	w25q_writeEnable();                  	//SET WEL 
	W25Q_CS_LOW();                            	//使能器件   
	w25q_spi_readWriteByte(W25X_PageProgram);   //发送写页命令   
	if(W25QXX_TYPE==W25Q256)                	//如果是W25Q256的话地址为4字节的，要发送最高8位
	{
			w25q_spi_readWriteByte((uint8_t)((WriteAddr)>>24)); 
	}
	w25q_spi_readWriteByte((uint8_t)((WriteAddr)>>16)); //发送24bit地址    
	w25q_spi_readWriteByte((uint8_t)((WriteAddr)>>8));   
	w25q_spi_readWriteByte((uint8_t)WriteAddr);   
	for(i=0;i<NumByteToWrite;i++)w25q_spi_readWriteByte(pBuffer[i]);//循环写数  
	W25Q_CS_HIGH();                            
	w25q_waitBusy();					   		//等待写入结束
}
//无检验写SPI FLASH 
//必须确保所写的地址范围内的数据全部为0XFF,否则在非0XFF处写入的数据将失败!
//具有自动换页功能 
//在指定地址开始写入指定长度的数据,但是要确保地址不越界!
//pBuffer:数据存储区
//WriteAddr:开始写入的地址(24bit)
//NumByteToWrite:要写入的字节数(最大65535)
//CHECK OK
void w25q_writeNoCheck(uint8_t* pBuffer,uint32_t WriteAddr,uint16_t NumByteToWrite)   
{ 			 		 
	uint16_t pageremain;	   
	pageremain=256-WriteAddr%256; //单页剩余的字节数		 	    
	if(NumByteToWrite<=pageremain)pageremain=NumByteToWrite;//不大于256个字节
	while(1)
	{	   
		w25q_writePage(pBuffer,WriteAddr,pageremain);
		if(NumByteToWrite==pageremain)break;//写入结束了
	 	else //NumByteToWrite>pageremain
		{
			pBuffer+=pageremain;
			WriteAddr+=pageremain;	

			NumByteToWrite-=pageremain;			  //减去已经写入了的字节数
			if(NumByteToWrite>256)pageremain=256; //一次可以写入256个字节
			else pageremain=NumByteToWrite; 	  //不够256个字节了
		}
	};	    
}

//擦除整个芯片		  
//等待时间超长...
void w25q_eraseChip(void)   
{                                   
	w25q_writeEnable();                  //SET WEL 
	w25q_waitBusy();   
	W25Q_CS_LOW();                            //使能器件   
	w25q_spi_readWriteByte(W25X_ChipErase);        //发送片擦除命令  
	W25Q_CS_HIGH();                            //取消片选     	      
	w25q_waitBusy();   				   //等待芯片擦除结束
}   
//擦除一个扇区
//Dst_Addr:扇区地址 根据实际容量设置
//擦除一个扇区的最少时间:150ms
void w25q_eraseSector(uint32_t Dst_Addr)   
{  
	//监视falsh擦除情况,测试用   
 	//printf("fe:%x\r\n",Dst_Addr);	  
 	Dst_Addr*=4096;
	w25q_writeEnable();                  //SET WEL 	 
	w25q_waitBusy();   
	W25Q_CS_LOW();                            //使能器件   
	w25q_spi_readWriteByte(W25X_SectorErase);   //发送扇区擦除指令 
	if(W25QXX_TYPE==W25Q256)                //如果是W25Q256的话地址为4字节的，要发送最高8位
	{
			w25q_spi_readWriteByte((uint8_t)((Dst_Addr)>>24)); 
	}
	w25q_spi_readWriteByte((uint8_t)((Dst_Addr)>>16));  //发送24bit地址    
	w25q_spi_readWriteByte((uint8_t)((Dst_Addr)>>8));   
	w25q_spi_readWriteByte((uint8_t)Dst_Addr);  
	W25Q_CS_HIGH();                            //取消片选     	      
  w25q_waitBusy();   				    //等待擦除完成
}  

